package com.cuz.daileetcode;

import com.cuz.daileetcode.utils.ArrayUtils;

/****
 * 快速排序
 */
public class Day24_5 {
    public static void main(String[] args) {
        int[] arr = {3, 2, 1, 2, 7, 8, 6, 6, 6, 6, 9, 0};
        quickSort(arr);
        ArrayUtils.print(arr);
    }

    public static void quickSort(int[] arr) {
        if (arr == null || arr.length < 2) {
            return;
        }
        quickSort(arr, 0, arr.length - 1);
    }

    public static void quickSort(int[] arr, int l, int r) {
        if (l < r) {
            int randomIndexBetweenLR = l + (int) (Math.random() * (r - l + 1));
            System.out.printf("当前选到的位置%s\n", randomIndexBetweenLR);
            System.out.printf("flag%s\n", arr[randomIndexBetweenLR]);
            System.out.println("交换前");
            ArrayUtils.print(arr);
            ArrayUtils.swap(arr, r, randomIndexBetweenLR);
            System.out.println("交换后");
            ArrayUtils.print(arr);
            int[] partition = partition(arr, l, r);
            System.out.println("部分有序");
            ArrayUtils.print(arr);
            System.out.println("区间");
            ArrayUtils.print(partition);
            System.out.println("区间的值");
            ArrayUtils.print(arr,partition[0],partition[1]);
            //递归让 让左半部分，右半有序
            quickSort(arr, l, partition[0] - 1);
            quickSort(arr, partition[1] + 1, r);
        }
    }

    /****
     * 划分区域
     * 让l，r返回的数满足
     * xxxx（小于arr[r]）xxx(等于arr[r])xxxx 大于arr[r]
     * @param arr 需要被划分的数组
     * @param l 左边界
     * @param r 有边界
     * @return new int[]{小于arr[r]的区域有边界，大于arr[r]的左边界}
     */
    public static int[] partition(int[] arr, int l, int r) {
        //小于flag数的右边界
        int lessThan = l - 1;
        //大于flag 数的左边界
        int moreThan = r;
        int flag = arr[r];
        //当前遍历到1位置
        int index = l;
        //如果当前的位置小于【大于flag 数的左边界】表示还存在数没有被遍历
        while (index < moreThan) {
            //小于标志 放在小于1区间 区间++ index++
            if (arr[index] < flag) {
                ArrayUtils.swap(arr, index, lessThan + 1);
                //区间有移动表示这个数位于小于区间
                lessThan++;
                //表示这个数被处理过了
                index++;
            } else if (arr[index] == flag) {
                //如果相等 index++，区间不做任何改变
                index++;
            } else {
                //和大于区间边缘数交换 表示放到大于区间
                ArrayUtils.swap(arr, index, moreThan - 1);
                //大于区间左移 表示 这个数位于大于区间
                moreThan--;
                //这里不移动index，因为大于区间移动的数我们还没有处理过
            }
        }
        //2,4,5,3,7,9,8 [6]
        //flag是6 和 7（moreThan对应的值）交换
        //2,4,5,3,6,9,8,7
        ArrayUtils.swap(arr, moreThan, r);
        return new int[]{lessThan + 1, moreThan};
    }
}
